import os
import sys
from fastmcp import FastMCP
from server.excel_analyse import excel_analyse, get_excel_header, get_excel_sheet_names

# 获取当前脚本的路径
script_path = os.path.abspath(sys.argv[0]) if sys.argv[0] else '__main__'  # 如果是在交互模式下，使用'__main__'作为脚本名

# 获取脚本所在的目录
app_path = os.path.dirname(script_path)

# 实例化通用服务
server = FastMCP(
    name="excel_analyse",
    instructions='Excel数据统计分析'
)


@server.prompt("echo")
def greeting(text: str) -> str:
    """创建一个个性化的问候提示。

    Args:
        customer_name: 客户名称

    Returns:
        包含问候消息的提示
    """
    return "这是一个mcp提示词"

# 向server中添加工具：员工请假
server.add_tool(
    excel_analyse,
    description="Excel表格内数据分析",
    tags={'Excel分析'})

server.add_tool(
    get_excel_header,
    description="获取Excel表头",
    tags={'获取Excel表头'})

server.add_tool(
    get_excel_sheet_names,
    description="获取Excel中所有 sheet名称",
    tags={'sheet_name'})

if __name__ == "__main__":
    # server.run(transport="stdio")
    # 启动sse 服务
    server.run(transport="sse", port=8000, host="0.0.0.0")